import os, numpy as np

def gen_eye_fixed():
    B,R,C,K = 8,1024,1024,0
    golden = np.zeros((B,R,C), dtype=np.float32)
    idx = np.arange(R)
    golden[:, idx, idx + K] = 1.0
    os.makedirs("output", exist_ok=True)
    golden.tofile("./output/golden.bin")
    print("golden shape:", golden.shape, "dtype:", golden.dtype)

if __name__ == "__main__":
    gen_eye_fixed()